package models

import "github.com/beego/beego/v2/client/orm"

type RoleMenu struct {
	BModel
	RoleID uint64 `json:"role_id" orm:"column(role_id)"`
	MenuID uint64 `json:"menu_id" orm:"column(menu_id)"`
}

func NewRoleMenu() *RoleMenu {
	return &RoleMenu{}
}

func (tb *RoleMenu) TableName() string {
	return RoleMenuTable
}

func (tb *RoleMenu) GetByRole(roleID uint64) (lst []RoleMenu, err error) {

	db := orm.NewOrm()
	_, err = db.QueryTable(tb).Filter("role_id", roleID).All(&lst)
	return
}
